Added inheritance for the view controllers