Nit: please add `@param` documentation comments.