pub struct TaskState {
    pub previous_energy_usage: u64,
    pub budget: u64,
    pub parent: u32,
}