Update code to utilize Prism