Clean up and add TODOs.