Fix and add docstring to method juju.ensure_num_units