modify code
add test case