As discussed, maybe some comments or docstring to make it clearer what `client` is.