Added plotting to retrieval