progress so far need to make a better training model