Add time_budget