Adding unit tests
Merge pull request #4 from mmeyer2k/tests

Adding unit tests