Adding test coverage.
Merge pull request #3 from ricomoss/adding_tests

Adding test coverage.