Adding to the plotting functionalituy