Cleaned up the new plots code and created the supplementary materials version