Added documentation and cleaned up unused imports