Challenge1 in Python tested working