changed the key distribution for rebalance.py
Some improvements to the rebalance scenario