Added Momentum training implementation, gradient calculation and model update.