Cleaned up TODO.