Adjust assign_household again