- Removed unused imports.
- Add code comments