Add missing imports from previous PR
Merge pull request #5 from sealevelresearch/add-missing-imports

Add missing imports from previous PR