Fixing up docs