added train and test sets used for grading