Update train_and_predict.py