Cleaned up import statement