Add docstring in models module