Implement Neural Memory Update with Surprise and MomentumDeep-ML | Problem