Fix typo
Add TODOs