Adding screenshots feature