Add customization note
Fix docstring