Added current view controller to the layout protocol