Add websocket logging