refactoring.

add test codes.