q2_train fixed