updates for generalization stuff