Update ViewController.swift

Working ViewController.swift file