Updated docstring and removed shebang