Update demo with smaller code