Updated docstring and added output