modifed and switched train/test datasets