1 additional line to main class