diff --git a/dataloader.py b/dataloader.py
index 165e89c7bd0e76a45232eeaaabd8cc7d1b279ad5..9bfeb25558f8d8c1baaaa59236bdaa3cca5c517d 100644
--- a/dataloader.py
+++ b/dataloader.py
@@ -195,12 +195,13 @@ class Intentsity_Dataset(Dataset):
         return torch.tensor(self.seq[idx]), torch.tensor([self.energy[idx]]).float(), torch.tensor(
             self.precursor_charge[idx]), torch.tensor(self.intensity[idx]).float()
 
-storage = H5ToStorage('database/traintest_hcd.hdf5')
-storage.make_npy_file('data/intensity/sequence_train.npy','sequence_integer')
-storage.make_npy_file('data/intensity/intensity_train.npy', 'intensities_raw')
-storage.make_npy_file('data/intensity/precursor_charge_train.npy', 'precursor_charge_onehot')
-
-storage = H5ToStorage('database/holdout_hcd.hdf5')
-storage.make_npy_file('data/intensity/sequence_holdout.npy','sequence_integer')
-storage.make_npy_file('data/intensity/intensity_holdout.npy', 'intensities_raw')
-storage.make_npy_file('data/intensity/precursor_charge_holdout.npy', 'precursor_charge_onehot')
+if __name__ =='__main__' :
+    storage = H5ToStorage('database/traintest_hcd.hdf5')
+    storage.make_npy_file('data/intensity/sequence_train.npy','sequence_integer')
+    storage.make_npy_file('data/intensity/intensity_train.npy', 'intensities_raw')
+    storage.make_npy_file('data/intensity/precursor_charge_train.npy', 'precursor_charge_onehot')
+
+    storage = H5ToStorage('database/holdout_hcd.hdf5')
+    storage.make_npy_file('data/intensity/sequence_holdout.npy','sequence_integer')
+    storage.make_npy_file('data/intensity/intensity_holdout.npy', 'intensities_raw')
+    storage.make_npy_file('data/intensity/precursor_charge_holdout.npy', 'precursor_charge_onehot')