I agree. If they're an imminent danger to others, lock them up strictly for segregation and not gratuitous vengeance (that is, be as nice to them as practical given that you're locking them up), if they're manageable, leave them be.