Added docs and few more tests