python use enumerate in for loop