Changed main.java and added test.java