add plots for problem1