add to class?