Challenge 6 in Python. Now we are talking