Minor fixes
Rename function balance to getBalance