modify analyze.py