Added the updated generator file that I forgot in the previous commit