Update server.js - store socket object instead of socket.id