added more tests
Merge pull request #29 from kotlinski/fix/add-tests

Fix/add tests