Make plotting functions composable

Use any approach that allows the caller to compose the plots together without side-effects. This should be done without requiring calling code to add boilerplate. Typically one passes the axes to the plotting function. If no axes are passed, then it just creates a new figure or something like that.