Trainer.py for easier training...