Separated out the plotting Class