implementing for loop in python