Add state preview to demo