Fixed some bugs in plotting function #11

Closed emgthomas closed 4 years ago