Fixing more issues with docs