Continued work on train method. Added data set class