Minor bug in problem 7