Test suite and improving the docs