Add Power_Source method to train.rb