Working on with the text generator code