Trying to make the generator works