Add plot generator using matplotlib