Modified example to rework input