Update analysis.py