Updated solution to problem 3.39