Add balance function to Account.rb