Python Schelling Model Code