replace .class with .is_a?