Update prettify.py