Fixed code errors. Add TODO.