Conway's game of life in python with pygame