Added tests

As mentioned in the Readme I should provide some tests for features. No now I got the tests, next PR will be a feature :checkered_flag: