Add document selection logic
