Clean up docs wip