Added a class to unify the testing / training of models and abstract data point prediction from the user, to allow us to easily integrate and modify the code