Simplify SequenceOf.take implementation

Take advantage of GeneratorOf and remove an extra class.