Added generator ui