Need to update docstring to include bz