fixed some imports and removed version module.

I keep making this silly mistake.