Create websocket.py