Add balance.rb