Clean up TODOs