adding a docstring of the group argument to 'balanceplot'