Bringing back getBalance method