Problem 4 Done

Need tests for Q3 and Q4