Added partial testing for plots