Update capacity_planning.py