Added the button for when the view controller is replaced.