Add listener support to the graceful shutdown handler