Updated Import Class

use import method from AIO