Just some changes to documentation in utils.py