little tweek to Main.java