removed print statements, fixed images
added trainmodel images