update main.py in client