updated the TODO section