Count a class with sets | StemPy Lab