Add missing TODO
Remove unused code