Update keg.py