add test code
refactor the code