UWB Horn antenna with lens.

hello ,

I'm trying to plan an UWB horn antenna , i already created the the horn antenna in CST but i want to improve its Directivity i would like to add a lens to the antenna but i don't know how to create it via the lens in the horn by CST..
Would appericiate any advices.