could be better at generalization tho