Updated import statement