Fix up docs