Problem 3 Done

Need to add more tests