Updated neural_network to use the Theta cell instead of Theta1, Theta2.