updated web socket code