improving the setup to the pbr model