Add getBalance method