add test codes
update test code