Inputted main changes to main.java

TODO: Make code more "readable"