Improvements to documenation