Improve documentation and small formatting fixes