Add power management