Add docstring
Update test
v0.6.1