Remove trainer.py
Change directory structure