Add units

Need to consider units for the code