Demo tidy up