Game of life in Python