diff --git a/scripts/training/train_diach_emb.py b/scripts/training/train_diach_emb.py
index 154cdd2..441e11f 100644
--- a/scripts/training/train_diach_emb.py
+++ b/scripts/training/train_diach_emb.py
@@ -40,7 +40,7 @@
 from gensim.models import Word2Vec
 from tqdm import tqdm
 import timeit
-from .utils import stopwdsrm, cleantxt, smart_procrustes_align_gensim
+from utils import stopwdsrm, cleantxt, smart_procrustes_align_gensim
 
 # ------------------- Start timing the whole process --------------------
 
@@ -61,23 +61,21 @@
 skip_alignment = configs['alignment']['skip']
 
 ## -- Preprocessing
-if skip_preprocessing == False:
-    savepreprocessed = configs['preprocessing']['savepreprocessed']
-    minwordlength = configs['preprocessing']['pipelines']['minwordlength']
-    lowercase = configs['preprocessing']['pipelines']['lowercase']
-    remove_punctuation = configs['preprocessing']['pipelines']['remove_punctuation']
-    remove_stopwords = configs['preprocessing']['pipelines']['remove_stopwords']
+savepreprocessed = configs['preprocessing']['savepreprocessed']
+minwordlength = configs['preprocessing']['pipelines']['minwordlength']
+lowercase = configs['preprocessing']['pipelines']['lowercase']
+remove_punctuation = configs['preprocessing']['pipelines']['remove_punctuation']
+remove_stopwords = configs['preprocessing']['pipelines']['remove_stopwords']
 
 ## -- Training
-if skip_training == False:
-    epochs = configs['training']['options']['epochs']
-    vector_size = configs['training']['options']['vector_size']
-    sg = configs['training']['options']['sg']
-    min_count = configs['training']['options']['min_count']
-    window = configs['training']['options']['window']
-    start_alpha = configs['training']['options']['start_alpha']
-    end_alpha = configs['training']['options']['end_alpha']
-    workers = configs['training']['options']['workers']
+epochs = configs['training']['options']['epochs']
+vector_size = configs['training']['options']['vector_size']
+sg = configs['training']['options']['sg']
+min_count = configs['training']['options']['min_count']
+window = configs['training']['options']['window']
+start_alpha = configs['training']['options']['start_alpha']
+end_alpha = configs['training']['options']['end_alpha']
+workers = configs['training']['options']['workers']
 
 # ------------------- Create dirs if needed --------------------
 
@@ -200,7 +198,7 @@
     model1.save('./outputs/{}/aligned/{}'.format(namethetest,model1_name))
 
     # --- Aligned each of the models to the model for the last time slice
-    for model in allmodels[:-2]:
+    for model in allmodels[:-1]:
 
         model2_name = model.split('/')[-1]
         print('Now aligning {} to {}...'.format(model1_name,model2_name))