Improve import util function