Added a Main class