Make initializing the plot a separate step