use .is_a?() instead of .class ==