Added the support to add and remove flow of controller