Update config to take a block for docs logic