Fix Loop Work

previously this output just the initial state of the problem, not the work done. I'm pretty sure A was done correctly but not clearly. The last line needed to be changed to:

work = mu * b (cos(theta)-1)