Add TODO as a suggestion for refactoring.