Trying to add more plots, not really working, also adding some ideas on how to generate the full visualization automatically