TODOs

I updated the todos