Address 2 TODOs