Fixed create_train