Improving convergence; implementing momentum