Using is_a? instead of string comparison