Please add a docstring for new `public_key` argument.