Added some comment and docstring to test