Add docstrings since I'm making this code public :)