First predict the value then training. Got rid of many unnecesary code.