Update ViewController.swift

Made edit on github