Add document selection logic