Add momentum and plot the error and standard deviation for each step