Add a docstring and fix the import.