add tests
Merge pull request #3 from sonots/add_test

Added tests