Momentum-Based Parameter Updater - Coding problem | OneNoughtOne