DOC: Fix up docs