add test to problem2