Update lib/web_socket.rb