Updated example according to latest changes