MAINT: Fixed a few docstrings and cosmits