It seems the problem is overfitting