add balance code