Update a util.py which make small files for quick test