Update MainViewController.swift