updating import statements to match separate module