Add tests
Merge pull request #2 from bakura10/tests

Add tests