Add gitignore
Add bare-bones code for training
PEP8
Add comments