Simple implementation of the game of life in PyGame