Solved Problem 4
Fixed the solution to problem 4