Fixing up documentation further