CLEANUP: corrected some docstrings about numpy ndarray shapes