Added disconnect handling

Comet is a little slow, but it works..