Challege5 in Python