Create schelling.py