Removed unused imports and updated the junit dependency version.