Add references section
Refine docstring