Update demo slightly