Removed unnecesary print and added docstring