- Refactored code.
- Removed imports.