Update comments.

Remove TODOs.