Cleaning up TODOs
Relocating TODOs