Update viewer.py