Game of life implementation in python with pygame.