Remove unused function. Update docstrings