Added the class and the main method