Update main_ab.py