diff --git a/tools/warn.py b/tools/warn.py
index dc6aacd..27a2695 100755
--- a/tools/warn.py
+++ b/tools/warn.py
@@ -2333,10 +2333,14 @@
   print '{},,{}'.format(total, 'All warnings')
 
 
-##### Main function starts here. #########################
+def main():
+  parse_input_file()
+  if args.gencsv:
+    dump_csv()
+  else:
+    dump_html()
 
-parse_input_file()
-if args.gencsv:
-  dump_csv()
-else:
-  dump_html()
+
+# Run main function if warn.py is the main program.
+if __name__ == '__main__':
+  main()