Update main_analysis.py