add plotting to task2