add test for exercise 4