updated tests, problem3, added screenshot