Add test code
Merge pull request #12 from sue445/feature/test_unit

Add test code