use proper data set, overfitting found