Fix sufficient balance bug + minor tweaks.