cleaning up import function

Signed-off-by: noahg <noah.gunther@gmail.com>