Finishing problem 5.1 although the graph seems a bit strange to me.