Add test
Refactor
Merge pull request #2 from sue445/feature/add_test

Add test and refactor