fixing problem with viewController