Add power source method to Train class