Updated import method