Clean up and TODOs.