Add handler for client disconnection