add annotation for main function

src/main/scala/leon/Annotations.scala

@@ -3,4 +3,5 @@ package leon
 object Annotations {
   class induct extends StaticAnnotation
   class axiomatize extends StaticAnnotation
+  class main extends StaticAnnotation
 }

src/main/scala/leon/TestGeneration.scala

@@ -19,6 +19,8 @@ class TestGeneration(reporter: Reporter) extends Analyser(reporter) {
   def analyse(program: Program) {
     z3Solver.setProgram(program)
     reporter.info("Running test generation")
+    program.definedFunctions.foreach(fd => println(fd.annotations))
+
     val testcases = generateTestCases(program)
 
     val topFunDef = program.definedFunctions.find(fd => fd.body.exists(body => body match {

src/main/scala/leon/plugin/CodeExtraction.scala

@@ -203,6 +203,7 @@ trait CodeExtraction extends Extractors {
               a.atp.safeToString match {
                 case "leon.Annotations.induct" => funDef.addAnnotation("induct")
                 case "leon.Annotations.axiomatize" => funDef.addAnnotation("axiomatize")
+                case "leon.Annotations.main" => funDef.addAnnotation("main")
                 case _ => ;
               }
             }