Add cleanup code on socket disconnect