Added screenshot saving