Add a main class to run the program
Edit main output