Cleaned up ViewController code