Added Main.class for tests