Fixed exit
Fixed an import bug