updated solution to group challenge