Move websocket.py to contrib directory