adding in tests
Merge pull request #7 from zachseifts/master

adding in tests