Tidy up demo