Update ViewController.swift

suggestions by Sidney de Koning  ;)