Improve TODO