WIP: Make client handle connection loss better.