added a main class for running the program