Updated TODO, changed main to be able to run iteratively.