model is wrong

need to add Stochastic gradient descent and Rum