Write a generator | StemPy Lab