Remove unneeded imports

Closes #46