Add sorting on nested associations