Initial tests
More tests
Merge pull request #5 from nickb1080/test

Add tests