Fix: Add docstring