New functionalities for momentum term; scaling lr

New functionalities for momentum term; added scaling for the learning
rate