Change docstrings and setup