Add method get_type