Units library

Would be cool to use something like Pint for inputs and outputs, to simplify things and make them more clear. There is `pint-pandas` that I can try for the output.

Maybe could be optional