Fixed problem 4

Realized I wasn't answering the question that was being asked