Fixed docstring
Updated comment
Addressed test comments