Updated demo to auto create the highlighted code blocks