Issue #8 Add unit test to the pull request.