Add more docstrings, improve naming a bit