Check battery level Zwave and Zigbee devices

I did install this flow :
https://www.reddit.com/r/homeassistant/comments/czqrk0/battery_check_for_ha_using_nodered/
Working fine and creating an HTML file giving an battery level over view.
It is checking on: attributes.battery_level
The problem is this is only showing my Zwave devices.
Is it possible to add my Zigbee devices also at this flow somehow?
(I did read somewhere to check on device_class: battery)
I do not know how to implement this into the zwave flow.
Any advise?

The answer is allready given in this threat: