3/11 fixed bugs in data preprocessing step with labels and train/test split