Just changed the example Main class