AI enhanced Fig 2