add test case
update the code