cleaned up update.py