tweaked withdraw method to check balance
a few tweaks to the balance check on withdrawal