Fix up documentation