Mobs Immune to Arrows

A feature to make mobs immune to arrows, it make arrows bounce on them, like in Wither Phase 2

Here a piece of code someone from Unofficial MCreator Discord provided

public boolean attackEntityFrom(DamageSource source, float amount)
{
    Entity entity = source.getImmediateSource();
    if (entity instanceof EntityArrow)
    {
        return false;
    }
    else
    {
        return super.attackEntityFrom(source, amount);
    }