Fix docs to work with new restructuring