Clean up a 'few' TODOs