diff --git a/build.gradle b/build.gradle
index 563d2b3..d68a5ea 100644
--- a/build.gradle
+++ b/build.gradle
@@ -11,8 +11,8 @@ plugins {
     id 'java-library'
 }
 
-//apply plugin:'application'
-//mainClassName = 'hgu.csee.isel.alinew.szz.AgSZZ'
+apply plugin:'application'
+mainClassName = 'hgu.csee.isel.alinew.szz.AGSZZRunner'
 
 repositories {
     // Use jcenter for resolving dependencies.