Organize answer of pythonchallenge.com