Add power management script