Updated TODOs a little