Added screenshot integration