Problem 4
Code