Small update to view controller