Fixed the parameters around to work with the Main.java