[WIP] Started adding docstrings and cleaning up.