better solution to import a class