Improvement to screenshot widget