Add power_source method to Train class