trying to make plots into a class