Add some docstrings and comments