changed draw code to generator