Updated the TODO section