Update updater.py

I think this modification could address issue #26 completely.