Added action in view controller to control what happens when you tap button