modify source code.
add test logic.