Modified README and removed train_data.py