Fourier Neural Operator based surrogates for $CO_2$ storage in realistic geologies