Docstring improvement: include example and rephrase
Cleanup: Only import if necessary