Problem 3 work

Started writing test data for problem three