Adding Unit definition to the python modul