Python implementation of Schelling segregation model