Adding logic to save the party members