Update units.rst

fix code samples