How do I make a mob similar to the Wither Storm with its growths being an attribute?

So, I'm making a mod based on the Lego Storm, which is a Pokemon concept I made and I want its phases to be an attribute that determines its phase number and model. How do I do that?