add balance.py