added better training and testing data