Python The Hardway code