Update pretty.py