add power_source method inside of train.rb