Added class for plotting data