Slight cleanup of Union and docstring.