Adding MDArray integration