Modify utils.py
Merge pull request #6 from dongweiming/modify_utils

Modify utils.py