Add import for modules