Added tests
Updated readme