Upgraded example with a better UI