How to render a block that my entity picks up?

[iProcellarum]

So I've created this entity that picks up blocks by basically copying and pasting the stuff from the EntityEnderman class. It works perfectly and it picks up blocks in the world and places them back down.

 

HOWEVER, it does not render the block that it picked up. The block just disappears from the world when the entity takes it, and reappears when the entity places it back down. I want the block it picked up to appear, similar to how blocks appear in an Enderman's hands. I cannot find anything to do with block models in EntityEnderman.class or ModelEnderman.class.

The ModelEnderman class does have some lines about positioning the Enderman's hands when it's carrying a block, but nothing about the block itself.

 

I'm not really sure where to look, so any help would be appreciated.

I'm doing this in 1.12.2 by the way.

Edited March 16, 2020 by iProcellarum