Removed some unnecessary code in Main.java