Update balanceplot.py

Use another to draw