TODOs Clean up