add test for question 2