Support staff locking on unit page

STUD-1873
@benmcmorran I'll add a helper as I agree the code is a little hard to read at the moment.