big mess of stuff
changed Momentum scheme to accept layerwise and componentwise parameters