Fixed terrible bug in balance update function
Improved explanations