Added screenshot functionality