Update main_v2.py