Update einstein.py