Added socket.io setup