updated import statement, fixed for loop