Update poc to use badge