add tests for problem 5 and 6