Add power_source method to check what power source the train uses.