Added the initial code and main method