Add ability to return DMA code