Add power source method to train class