improve auto test for 2(b)
finish problem 3