Small update on viewController