Add support for dma