Update docs and docstrings
Remove accidental print satement