Weapon Icon (Small icon in the left corner) make that increase with an upgrade?

[X-maul]

I want to make the number in the corner of one of my units increase when I make an upgrade, how do I do that? I did set the Affected unit array to my unit, but that does nothing

What am I missing?

NOTE: I feel retarded, I know it's in the bottom right corner

 

[MissKerrigan]

it's in the upgrade stats, set the 'level' +1