Tidies up imports