Added plotting to part 3