Update ex3_main.py