Fix docs more