Game of life implementation using Python and Pygame