Take a stab at making a generator
Make generator work