Add some TODOs