We should set up a dictionary with plot functions so that we can easily extend the provided functionality.