Update watermark.py