Add TODO list
Fix output