Clean up TODO