Updated commenting to reflect changes in parameters, return types

This commit message summarizes the changes made to the code and
highlights the importance of updating docstrings and comments to keep
them consistent with the new code.