(cleanup) Updated example for new API.