add train model interface
Merge branch 'master' of https://github.com/PittCOE1186/COE1186FinalProject
add train mode;