Add TODO to main method