added:
1. checking for units
2. docstring
3. minor improvements