write the crack the code interview using python