add test and modified the result