Update Cellular automaton.py