Add simple game of life in python