Change model parameters and train for more steps for unit 1 challenge

initial commit