Add unit mapping to unitDto