Added generator test