Conway's Game of Life written in python using pygame