Fixed main method in main class to include test class