Update websocket.rb