Simplified demo by moving complex set up to a util function