Merge pull request #25 from jerv13/master

Update unit test