add plot generator for second lab