Add TODOs to fix