Updated container view controller to not use screen size