Added generator model