Adding Train class to let user to have their own SVM