Remove unused code
Add TODO