more updates to class plotting
more changes to class plotting