added weight decay and momentum