Update cellular_automata.py