Add method to set account balance.