Fixed up ViewController for final release.