[FEATURE] Fix comments, used imperative form for docstrings