Change to docs to use new callback style