Changed to use Nesterov Momentum

Compute momentum update before taking gradient and keep track of all
updates for new momentum term