Create balanced.py