use is_a? instead of .class ==