Of course there is much potential for improvements and your suggestion is really nice^^
Since this was just for a gamejam, I am not working on this game anymore, but maybe I will create a new and better version someday :)
If you like you can modify the game by yourself. All game data for the generators is basically defined in a textfile inside of the archive. So you can just open the game with a tool like WinRAR and then modify the generators.json file.