Add controller to switch betwen cases