Added some tests
Documentation updates