Add function to extract stimulus-specific network