Add some plotting to class 6