Update example for easier testing