Put account balance update in its own method