use enumerate in for loop python