Final fixes for demo.