Add get_roles method