check whether need screenshot or not
improve design