Remove unused imports
Bumped version to 2.0.4