updating the rolling upgrade example to use the new module