Improve docstrings

Change type hinting from int to float