Added some tests.
Merge pull request #28 from jgallen23/feature/tests

Added some tests