Balance now updates in logic as well as interface