Added controller for modal