Makes minor changes to in-class demo.