Add test problems
Modify test codes