finished stoch_grad_mom.m