Conway's Game of Life in Python using pygame