Adding documentation and fixing imports.