python for loop with index range