FIX PR comments: docstring, tests, and bug fixes