Add pbr support