can you switch to using `{{screen_id}}` for the token highlighting?