Implemented the remove model context button