Update initial_analysis.py