Add get_identity function