Clean up, add TODOs.