Add web socket support