Replaced == with is_a?