diff --git a/.gitignore b/.gitignore
index 4f21f62..d2afb40 100644
--- a/.gitignore
+++ b/.gitignore
@@ -242,4 +242,8 @@ paket-files/
 
 # JetBrains Rider
 .idea/
-*.sln.iml
\ No newline at end of file
+*.sln.iml
+
+# Custom
+media/
+logs/
\ No newline at end of file
diff --git a/scripts/WGAN-GP.py b/scripts/WGAN-GP.py
index 2bc9172..c9b953d 100644
--- a/scripts/WGAN-GP.py
+++ b/scripts/WGAN-GP.py
@@ -29,6 +29,16 @@ Fs = 16000
 DATA_DIR = r"D:\ML_Datasets\violin"
 INSTRUMENT = "violin"
 
+print("Creating necessary directories")
+
+paths = ["logs/train", 
+         f"models/{INSTRUMENT}/js",
+         f"output/{INSTRUMENT}",]
+
+for path in paths:
+    if not os.path.exists(os.path.join(os.getcwd(), path)):
+        os.makedirs(path)
+
 # Define class that contains GAN infrastructure
 class GAN:
     def __init__(self, model_dims=MODEL_DIMS, num_samples=NUM_SAMPLES,