diff --git a/examples/demo.py b/examples/demo.py
index ae2c872ed090f30fb628c0c9f1da108a9141710e..fb7196bdaef41ed36d2e190860678302209fae69 100644
--- a/examples/demo.py
+++ b/examples/demo.py
@@ -165,4 +165,6 @@ class Demo:
         demo_001.set_record_frames(os.path.join(__file_dirname__, '..', 'rendering', 'frame_{:04d}.bmp'))
         demo_001.run_demo(360)
 
-Demo.run_complex_scene()
+
+if __name__ == "__main__":
+    Demo.run_complex_scene()
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 791a07aaff3140fa89fc74945655d3720c5b270c..f644bc120d4b514f1c54e0330cfc8dc4654a4f4e 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -1,7 +1,7 @@
 import numpy as np
 
 from flatland.core.transition_map import GridTransitionMap
-from flatland.core.transitions import Grid8Transitions, RailEnvTransitions
+from flatland.core.transitions import RailEnvTransitions
 from flatland.envs.env_utils import distance_on_rail, connect_rail, get_direction, mirror
 from flatland.envs.env_utils import get_rnd_agents_pos_tgt_dir_on_rail
 
diff --git a/flatland/utils/graphics_pil.py b/flatland/utils/graphics_pil.py
index e6a92d3ddc71f278441ada683a8320d4a2ab3539..be804c0348cf0fca9cc0c089c64693f7a617064b 100644
--- a/flatland/utils/graphics_pil.py
+++ b/flatland/utils/graphics_pil.py
@@ -384,7 +384,6 @@ class PILSVG(PILGL):
         }
 
         # "paint" color of the train images we load
-        # a3BaseColor = self.rgb_s2i("0091ea")
         a3BaseColor = self.rgb_s2i("d50000")
 
         self.dPilZug = {}