Import modules with type definitions where possible

Use `import` where possible to import modules, along
with type info.