conditional and loops in python