Added some comments to websocket.py