Unified format for docstring.
Added import in the main.