remove weight decay argument from momentum update functions. Fixes #69.