using enumerate in python for loop