first three challenge in python