Cellular Automata

I built this as a project to help me learn python and pygame. It is a base implementation of the Game of Life routine. I am not entirely happy with the speed however, and would appreciate any feedback as to how I could optimize the code. I eventually want to work this into a puzzle game of sorts, but have to spend a little more time ironing out the details before I set up the actual game.