Game of life in pyGame