first attempt at training model 2