1 files changed, 6 insertions, 4 deletions

diff --git a/unit_trace/viz/visualizer.py b/unit_trace/viz/visualizer.py
index 5027e89..c0186f7 100755
--- a/unit_trace/viz/visualizer.py
+++ b/unit_trace/viz/visualizer.py
@@ -1,9 +1,11 @@
 #!/usr/bin/python
 
-import viewer
 import convert
+import viewer
 import renderer
+import schedule
 import format
+import pygtk
 import gtk
 
 TIME_PER_MAJ = 10000000
@@ -11,13 +13,13 @@ TIME_PER_MAJ = 10000000
 def visualizer(stream):
     sched = convert.convert_trace_to_schedule(stream)
     sched.scan(TIME_PER_MAJ)
-    
+
     task_renderer = renderer.Renderer(sched)
     task_renderer.prepare_task_graph(attrs=format.GraphFormat(time_per_maj=TIME_PER_MAJ))
     cpu_renderer = renderer.Renderer(sched)
     cpu_renderer.prepare_cpu_graph(attrs=format.GraphFormat(time_per_maj=TIME_PER_MAJ))
-    
+
     window = viewer.MainWindow()
     window.set_renderers({'Tasks' : task_renderer, 'CPUs' : cpu_renderer})
-    
+
     gtk.main()