Parametize main class
Set main class parameter