added tests
Merge pull request #6 from pirfalt/tests

added tests