Add TODOs and fix an error message