python foreach loop with index