added sample train and test sets