Pyrotastic Firework Model

package com.creativityshark.pyrotastic.client.entity.model;

import com.creativityshark.pyrotastic.PyrotasticMod;
import com.mojang.blaze3d.vertex.PoseStack;
import com.mojang.blaze3d.vertex.VertexConsumer;
import net.minecraft.client.model.EntityModel;
import net.minecraft.client.model.geom.ModelLayerLocation;
import net.minecraft.client.model.geom.ModelPart;
import net.minecraft.client.model.geom.PartPose;
import net.minecraft.client.model.geom.builders.CubeListBuilder;
import net.minecraft.client.model.geom.builders.LayerDefinition;
import net.minecraft.client.model.geom.builders.MeshDefinition;
import net.minecraft.client.model.geom.builders.PartDefinition;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.entity.projectile.FireworkRocketEntity;
import org.jetbrains.annotations.NotNull;

public class FireworkRocketModel extends EntityModel<FireworkRocketEntity> {
    public static final ModelLayerLocation MODEL_LAYER_LOCATION = new ModelLayerLocation(new ResourceLocation(PyrotasticMod.MOD_ID, "firework_rocket"), "base");

    private final ModelPart root;

    public FireworkRocketModel(ModelPart root) {
        super(RenderType::entityCutoutNoCull);
        this.root = root.getChild("base");
    }

    public static LayerDefinition createBodylayer() {
        MeshDefinition modelData = new MeshDefinition();
        PartDefinition modelPartData = modelData.getRoot();
        PartDefinition modelPartData2 = modelPartData.addOrReplaceChild("base", CubeListBuilder.create().texOffs(0, 0).addBox(-1.5F, -7.0F, -1.5F, 3.0F, 13.0F, 3.0F), PartPose.ZERO);
        modelPartData2.addOrReplaceChild("rim", CubeListBuilder.create().texOffs(0, 16).addBox(-2.5F, -5.0F, -2.5F, 5.0F, 2.0F, 5.0F), PartPose.ZERO);
        return LayerDefinition.create(modelData, 32, 32);
    }

    public void renderToBuffer(@NotNull PoseStack matrices, @NotNull VertexConsumer vertices, int light, int overlay, float red, float green, float blue, float alpha) {
        this.root.render(matrices, vertices, light, overlay, red, green, blue, alpha);
    }

    @Override
    public void setupAnim(@NotNull FireworkRocketEntity pEntity, float pLimbSwing, float pLimbSwingAmount, float pAgeInTicks, float pNetHeadYaw, float pHeadPitch) {

    }
}