Add 2 TODOs