Remove unused function
Add docstrings