scala for loop with index