Add missing functionality to disconnect client when server restarts