Refator the currentBalance() method in controller