Improve TODO description in comment