train on the small data set