Add callback for when the context menu is closed