Fix needs method and move highlighted families to a partial