Move import statement to main