Fixing some TODOs.