Adding support for 4D MDArrays