Fix comments, used imperative form for docstrings