Added minor doc.
Fixed boolean return
Reverted removal of unused imports.
Will hopefully make PR diff more tolerable.