Add plotting to CheckoutLineSimulation.py