add enter/exitEditMode functionality to the main Controller