Update to balance functions.