Python, easy, challenge 2 is completed