Update YRMainViewController.swift