Added __main__ run and docstring