added some more methods. added test and train data