Plug: Add a close handler