Add reconnect support