Add disconnect listener to prevent Heroku crashes