Updated docstrings to reflect new interfaces