Update checkbalance.py