Convert to batch api.
Remove TODO.