Added documentation
Removed unused imports