Work on the last exercise some more