refine drawBattery.py to for better pics