CLEANUP: more corrections on docstrings about numpy ndarray shapes