add tests

[jdugan1024]

Before we release this code it would be good to have a basic test for this code.

[AhmedElywa]

#7 here is our pull request for this issue