Add a new python code from Head First Python