added plotting to notebook