Add more docstring and TODO.