Added a better demo with util functions.