Added momentum to the weight updates