--- a/python/ir/function.py	Mon Aug 12 20:14:47 2013 +0200
+++ b/python/ir/function.py	Wed Aug 14 20:12:40 2013 +0200
@@ -1,10 +1,9 @@
-from .basicblock import BasicBlock
+from .basicblock import Block
 
 class Function:
     def __init__(self, name):
       self.name = name
-      self.bbs = []
-      self.entry = None
+      self.entry = Block('entry')
 
     def __repr__(self):
         return 'Function {0}'.format(self.name)
@@ -19,7 +18,15 @@
         bb.parent = None
 
     def getBBs(self):
-        return self.bbs
+        bbs = [self.entry]
+        worklist = [self.entry]
+        while worklist:
+            b = worklist.pop()
+            for sb in b.Successors:
+                if sb not in bbs:
+                    bbs.append(sb)
+                    worklist.append(sb)
+        return bbs
 
     def findBasicBlock(self, name):
         for bb in self.bbs: