Conway's game of life, using pygame