Update balance.rb