Updated the view controller file