Added tests
Updated documentation