Changes for screenshot taking