Cosmetic changes to generator