Add generalize TODO