add screenshot functionality