Conway's Game of Life in Python using Pygame