Fixed test
Merge branch 'master' of https://github.com/meylingtaing/COE1186.git
Train controller now uses updated model constructor