Add partial import code