Enchant Levels on Armor

https://imgur.com/a/PAPSYIh

I am trying to make this enchantment increase your max health by varying amounts per level, then also remove the max health when taken off. I've been stuck for hours, any help would be appreciated. I feel like there is an easy way to go about this that I'm glossing over.

The itemstacks are placeholders, they wouldn't work either way but I'm trying to figure out how to inherit the armor and then get into its data.