update screenshot with highlighted tokens