Fixing up the docs