made a couple updates to imports