Add time prediction to analysis.py