Add docstrings to the important functions. Right now, one has to go through the code to understand how to use it.