Update TODO to reflect recent changes