diff --git a/.gitignore b/.gitignore
index f34729b..43817f2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
+/.idea/*
 /results/*
 /data/
 # Created by https://www.gitignore.io/api/data,linux,macos,python,pycharm,windows,jupyternotebooks
diff --git a/models/layers.py b/models/layers.py
index ab012f8..a346bc1 100644
--- a/models/layers.py
+++ b/models/layers.py
@@ -25,3 +25,4 @@ if __name__ == '__main__':
     X = np.random.rand(1, 60, 40)
     Y = sub_spec_tnfm(X)
     print(f'\t Sub-Spectrogram transformation from shape {X.shape} to {Y.shape}')
+    print('Done ...')
\ No newline at end of file