updated tests
Merge pull request #13 from iambernie/tests

updated tests