updates to class plotting