Remove TODO.
Add update functionality.