Stil beta, creating train method using gradient descent. Theta is now vertical vector.