Add power_source method