using enumerate in for loop python