Great idea and the mod works as intended. One thought though: I feel like shock and lightning should stay unchained, since they're basically a weaker and stronger version of the same power. I'm pretty sure they even cost the same number of FP, so there's not really a point to use shock when you have lightning. The rest make sense however since push, whirlwind, wave, and storm are all functionally and visually different.
Also, you should consider making a TSL version!