diff --git a/.gitignore b/.gitignore
index ac08111..82977a3 100644
--- a/.gitignore
+++ b/.gitignore
@@ -139,3 +139,4 @@ Temporary Items
 
 outputs/*
 inputs/mnist
+inputs/audio*
diff --git a/speech_data.py b/speech_data.py
index 5ccbeb6..a7cb35c 100644
--- a/speech_data.py
+++ b/speech_data.py
@@ -56,13 +56,14 @@ def sunflower_pairs_data():
     te_pairs = np.array([x_pos_test,x_neg_test]).reshape(x_pos_test.shape[0],2,max_samples,sample_size)
     return tr_pairs,te_pairs,tr_y,te_y
 
-def create_spectrogram_data():
+def create_spectrogram_data(audio_group='audio'):
     audio_samples = pd.read_csv('./outputs/'+audio_group+'.csv',names=['word','voice','rate','variant','file'])
+    # audio_samples = audio_samples.loc[audio_samples['word'] == 'sunflowers'].reset_index(drop=True)
     audio_samples.loc[:,'spectrogram'] = audio_samples.loc[:,'file'].apply(lambda x:'outputs/'+audio_group+'/'+x).apply(generate_aiff_spectrogram)
-    audio_samples.to_pickle('spectrogram.pkl')
+    audio_samples.to_pickle('outputs/spectrogram.pkl')
 
-def speech_pairs_data(audio_group):
-    audio_samples = pd.read_pickle('spectrogram.pkl')
+def speech_pairs_data(audio_group='audio'):
+    audio_samples = pd.read_pickle('outputs/spectrogram.pkl')
     y_data = audio_samples['variant'].apply(lambda x:x=='normal').values
     max_samples = audio_samples['spectrogram'].apply(lambda x:x.shape[0]).max()
     sample_size = audio_samples['spectrogram'][0].shape[1]