Modified some part (Step 2 and 3) of 'WebSocket Client'