* updated code.
* fixed test case.