Pretty conceptually simple to just walk the parts tree and find the ModuleAvionics that are remaining and sum up their massLimits and push it into StageStats, it just needs to be done with some reflection.