Added a main method!