Add plot generation