Minor changes in main : added TODO comment and removed useless line.