Updated key.py