Use ws instead of socket io