Add client socket code