Add get_id function