Added a setter to test if generation is working