Merge pull request #3 from jpsowers/master

Cleaning unit test class