Update getKey() in Main.java