diff --git a/src/diana/diana.py b/src/diana/diana.py
--- a/src/diana/diana.py
+++ b/src/diana/diana.py
@@ -1,13 +1,13 @@
 import os
 import re
 
-import config as cfg
-import go
-from go import BLACK,WHITE,EMPTY
-from sgfParser import ParserError
-from sgfParser.collection import Collection
-from drawer.svg import Svg
-from drawer.tikz import Tikz
+from . import config as cfg
+from . import go
+from .go import BLACK,WHITE,EMPTY
+from .sgfParser import ParserError
+from .sgfParser.collection import Collection
+from .drawer.svg import Svg
+from .drawer.tikz import Tikz
 
 
 def collectMoves(root):
@@ -116,7 +116,7 @@ W: {PW} {WR}
 		return True
 
 
-if __name__=="__main__":
+def main():
 	cfg.parseArgs()
 	print("processing:")
 	files=cfg.inputFiles[:]