Changes for balance update