added todo in Main.java'