Import logic for code gen