Make the demo nicer