Train data better again