Added server disconnect handling in client.