Improving docstrings, comments, README