did challenge 3 in python