Update Main.java

Made it so NearestNeighbor AI can be can be used.