Added four training sets for models 2,3,4,5