A simple WebSocket training script