adding proper account balance warning back in