Lawsuit: Tennessee private prison should improve conditions or be shut down