given the number of locations in psi domain can you switch this to use iter_docs?