Yeah, you are right. The diagram step 3 should also apply balance changes at the same time as adding transaction to the list. Basically adding transaction to the list is just a sign that some changes were applied to the balance.
btw in the link to original gist you can find more complete executable solution, with rollback code: https://gist.github.com/avsej/3136027