Update docs to match the latest changes