Completed problem 4. Need to refactor to method signature though.