Add (broken) support for plotting incorrect classifications