Add the ability to customise the template for demo