1st Step: Add set_role function