modified to pass unit test
Merge pull request #2 from takatama/unit-test

modified to pass unit test