Add doctest for Unit