Made some functions to make it easier to switch between train and test
Merge branch 'master' of https://github.com/TLadd/SeniorDesign