Update battle.rb