Add reconnection for socket.io