Updated docstring for problem 24