add screenshot generation