Tidy ups for the demo.