Test code in Main.java