added unit tests
added more unit tests
Merge pull request #1 from shaneob/unit-tests

Unit tests