Added convenience class for plotting
Exercise done