Modified ViewController for demo