Add UKI generation