Update documentation to new format for our count argument